import torch
from torch import nn


class Wangqi(nn.Module):
    def __init__(self, *args, **kwargs) -> None:
        super().__init__(*args, **kwargs)

    def forward(self,input):
        output=input+1
        return output

wangqi=Wangqi()
x=torch.tensor(1)
output=wangqi(x)
print(output)